pyspark interview question